#include <bits/stdc++.h>

using namespace std;

struct TreeNode
{
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode() : val(0), left(nullptr), right(nullptr) {}
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
    TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
};

class Solution
{
public:
    vector<vector<int>> levelOrder(TreeNode *root)
    {
        vector<vector<int>> res;
        queue<TreeNode *> que;
        if (root != nullptr)
        {
            que.push(root);
        }
        while (!que.empty())
        {
            int length = que.size();
            vector<int> vec;
            for (int i = 0; i < length; i++)
            {
                TreeNode *node = que.front();
                que.pop();
                vec.push_back(node->val);
                if (node->left)
                {
                    que.push(node->left);
                }
                if (node->right)
                {
                    que.push(node->right);
                }
            }
            res.push_back(vec);
        }
        return res;
    }

    // TreeNode *invertTree(TreeNode *root)
    // {
    //     queue<TreeNode *> que;
    //     if(root!=nullptr)
    //     {
    //         que.push(root);
    //     }
    //     while(!que.empty())
    //     {
    //         int length=que.size();
    //         for(int i=0;i<length;i++)
    //         {
    //             TreeNode* node=que.front();
    //             que.pop();
    //             swap(node->left,node->right);
    //             if(node->left)
    //             {
    //                 que.push(node->left);
    //             }
    //             if(node->right)
    //             {
    //                 que.push(node->right);
    //             }
    //         }
    //     }
    //     return root;
    // }

    // void traverse(TreeNode *cur)
    // {
    //     if (cur == nullptr)
    //     {
    //         return;
    //     }
    //     swap(cur->left, cur->right);
    //     traverse(cur->left);
    //     traverse(cur->right);
    // }

    // TreeNode *invertTree(TreeNode *root)
    // {
    //     TreeNode *cur = root;
    //     traverse(cur);
    //     return root;
    // }

    // TreeNode *invertTree(TreeNode *root)
    // {
    //     stack<TreeNode *> stk;
    //     if (root != nullptr)
    //     {
    //         stk.push(root);
    //     }
    //     while (!stk.empty())
    //     {
    //         TreeNode *node = stk.top();
    //         stk.pop();
    //         swap(node->left, node->right);
    //         if (node->right)
    //         {
    //             stk.push(node->right);
    //         }
    //         if (node->left)
    //         {
    //             stk.push(node->left);
    //         }
    //     }
    //     return root;
    // }

    // TreeNode *invertTree(TreeNode *root)
    // {
    //     stack<TreeNode *> stk;
    //     if (root != nullptr)
    //     {
    //         stk.push(root);
    //     }
    //     while (!stk.empty())
    //     {
    //         TreeNode *node = stk.top();
    //         if (node != nullptr)
    //         {
    //             stk.pop();
    //             if (node->right)
    //             {
    //                 stk.push(node->right);
    //             }
    //             if (node->left)
    //             {
    //                 stk.push(node->left);
    //             }
    //             stk.push(node);
    //             stk.push(nullptr);
    //         }
    //         else
    //         {
    //             stk.pop();
    //             node = stk.top();
    //             stk.pop();
    //             swap(node->left, node->right);
    //         }
    //     }
    //     return root;
    // }

    TreeNode *invertTree(TreeNode *root)
    {
        stack<TreeNode *> stk;
        TreeNode *cur = root;
        while (!stk.empty() || cur != nullptr)
        {
            if (cur != nullptr)
            {
                stk.push(cur);
                cur = cur->left;
            }
            else
            {
                cur = stk.top();
                stk.pop();
                swap(cur->left, cur->right);
                cur = cur->left;
            }
        }
        return root;
    }
};

int main()
{
    TreeNode *root = new TreeNode(4);
    root->left = new TreeNode(2);
    root->right = new TreeNode(7);
    root->left->left = new TreeNode(1);
    root->left->right = new TreeNode(3);
    root->right->left = new TreeNode(6);
    root->right->right = new TreeNode(9);
    Solution sol;
    TreeNode *res = sol.invertTree(root);
    vector<vector<int>> vec = sol.levelOrder(res);
    for (auto i : vec)
    {
        for (auto j : i)
        {
            cout << j << " ";
        }
        cout << endl;
    }
    return 0;
}